
MOT_root = '/home/sean/data/MOT/MOT17Det/'
write_file = '/home/sean/data/MOT/MOT17Det/ImageSet/train.txt'

train_video_list = ['train/MOT17-02', 'train/MOT17-04', 'train/MOT17-05', 'train/MOT17-09', 'train/MOT17-10',
                    'train/MOT17-11', 'train/MOT17-13']
train_video_len =[600, 1050, 837, 525, 654, 900, 750]

val_video_list = ['train/MOT17-01', 'train/MOT17-03', 'train/MOT17-06', 'train/MOT17-07', 'train/MOT17-08',
                    'train/MOT17-12', 'train/MOT17-14']

with open(write_file, 'w') as wf:
    for video, len in zip(train_video_list, train_video_len):
        for frame in range(1, len+1):
            wf.write(video+' '+str(frame).zfill(6)+'\n')
